package code.cando;

import javax.swing.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * author : yeswater
 * date : 2023/12/8
 * <p>
 * 字母异位词分组
 * 0959 - 1040
 */

public class T049 {
    public static void main(String[] args) {
        // String[] strs = {"eat", "tea", "tan", "ate", "nat", "bat"};
        // System.out.println(groupAnagrams(strs).toString());
        System.out.println('z' - 'a');
    }

    /**
     * 第一反应 麻烦的思路
     */
    public static List<List<String>> groupAnagrams1(String[] strs) {
        List<List<String>> res = new ArrayList<>();
        HashMap<int[], List<String>> map = new HashMap<>();
        List<int[]> list = new ArrayList<>();
        for (String str : strs) {
            int[] tool = new int[26];   // 字母组成
            boolean flag = false;       // 是否已有异位词
            for (int i = 0; i < str.length(); i++) {
                tool[str.charAt(i) - 'a'] += 1;
            }
            // 是否有异位词
            for (int[] arr : list) {
                if (Arrays.equals(arr, tool)) { // 有异位词
                    List<String> tmpList = map.getOrDefault(arr, new ArrayList<>());
                    tmpList.add(str);
                    map.put(arr, tmpList);
                    flag = true;
                }
            }
            if (!flag) {
                list.add(tool);
                ArrayList<String> newList = new ArrayList<>();
                newList.add(str);
                map.put(tool,newList);
            }
        }
        for (List<String> stringList : map.values()) {
            res.add(stringList);
        }
        return res;
    }

    /**
     * 题解机智的思路
     */
    public static List<List<String>> groupAnagrams(String[] strs) {
        HashMap<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String key = new String(chars);
            map.computeIfAbsent(key, k -> new ArrayList()).add(str); // map.computeIfAbsent() 返回 value
        }
        return new ArrayList<>(map.values());
        
    }


}
